#include <iostream>
#include <string.h>
#include <string>
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
//若没有正确输入命令行参数则提示用户输入参数
void Usage(const std::string& proc)
{
    std::cout<<"\n\rUsage: "<<proc<<" server ip,server port\n"<<std::endl;    
}

int main(int argc,char* argv[])
{
    if(argc!=3)                 //需传入三个命令行参数
    {
        Usage(argv[0]);
        exit(1);
    }
    //获取ip和端口
    std::string serverip=argv[1];
    uint16_t serverport=std::stoi(argv[2]);    

    int sockfd=socket(AF_INET,SOCK_STREAM,0);
    if(sockfd<0)
    {
        std::cerr<<"socket error"<<std::endl;
        return 1;
    }

    //连接目标服务端，填充服务端信息
    struct sockaddr_in server;
    memset(&server,0,sizeof(server));
    server.sin_family=AF_INET;
    server.sin_port=htons(serverport);
    inet_pton(AF_INET,serverip.c_str(),&(server.sin_addr));         //主机序列转网络序列


    //客户端发起connect，进行自动随机绑定
    int n=connect(sockfd,(struct sockaddr*)&server,sizeof(server));
    if(n<0)
    {
        std::cerr<<"connect error"<<std::endl;
        return 2;
    }

    //待发信息
    std::string message;
    while(1)
    {
        std::cout<<"please Enter# "<<std::endl;
        std::getline(std::cin,message);
        write(sockfd,message.c_str(),message.size());       //写（发）给服务端

        char buf[4096];
        int n=read(sockfd,buf,sizeof(buf));                 //读（收）服务端返回的信息
        if(n>0)
        {
            buf[n]=0;
            std::cout<<buf<<std::endl;                      //打印服务端返回的信息
        }
    }

    close(sockfd);  
    return 0;
} 